nlp_architect.procedures.transformers package

Submodules

nlp_architect.procedures.transformers.base module

nlp_architect.procedures.transformers.base.create_base_args(parser: argparse.ArgumentParser, model_types=None)[source]

Add base arguments for Transformers based models

nlp_architect.procedures.transformers.base.inference_args(parser: argparse.ArgumentParser)[source]

Add inference specific arguments for Transoformer based models

nlp_architect.procedures.transformers.base.train_args(parser: argparse.ArgumentParser, models_family=None)[source]

Add training specific arguments for Transformer based models

nlp_architect.procedures.transformers.glue module

class nlp_architect.procedures.transformers.glue.TransformerGlueRun(parser=None)[source]

Bases: nlp_architect.procedures.procedure.Procedure

static add_arguments(parser: argparse.ArgumentParser)[source]
static run_procedure(args)[source]
class nlp_architect.procedures.transformers.glue.TransformerGlueTrain(parser=None)[source]

Bases: nlp_architect.procedures.procedure.Procedure

static add_arguments(parser: argparse.ArgumentParser)[source]
static run_procedure(args)[source]
nlp_architect.procedures.transformers.glue.add_glue_args(parser: argparse.ArgumentParser)[source]
nlp_architect.procedures.transformers.glue.add_glue_inference_args(parser: argparse.ArgumentParser)[source]
nlp_architect.procedures.transformers.glue.do_inference(args)[source]
nlp_architect.procedures.transformers.glue.do_training(args)[source]
nlp_architect.procedures.transformers.glue.get_metric_fn(task_name)[source]

nlp_architect.procedures.transformers.seq_tag module

class nlp_architect.procedures.transformers.seq_tag.TransformerTokenClsRun(parser=None)[source]

Bases: nlp_architect.procedures.procedure.Procedure

static add_arguments(parser: argparse.ArgumentParser)[source]
static run_procedure(args)[source]
class nlp_architect.procedures.transformers.seq_tag.TransformerTokenClsTrain(parser=None)[source]

Bases: nlp_architect.procedures.procedure.Procedure

static add_arguments(parser: argparse.ArgumentParser)[source]
static run_procedure(args)[source]
nlp_architect.procedures.transformers.seq_tag.do_inference(args)[source]
nlp_architect.procedures.transformers.seq_tag.do_training(args)[source]
nlp_architect.procedures.transformers.seq_tag.process_inference_input(input_file)[source]

Module contents